"""
助手工具函数
"""

import datetime
import json

def format_output(data, format_type="json"):
    """
    格式化输出数据
    
    Args:
        data: 要格式化的数据
        format_type: 格式类型 ('json', 'text', 'pretty')
    
    Returns:
        格式化后的字符串
    """
    if format_type == "json":
        return json.dumps(data, ensure_ascii=False, indent=2)
    elif format_type == "text":
        return str(data)
    elif format_type == "pretty":
        if isinstance(data, dict):
            return "\n".join(f"{k}: {v}" for k, v in data.items())
        elif isinstance(data, list):
            return "\n".join(f"- {item}" for item in data)
        else:
            return str(data)
    else:
        raise ValueError(f"不支持的格式类型: {format_type}")

def get_timestamp(format_str="%Y-%m-%d %H:%M:%S"):
    """
    获取当前时间戳
    
    Args:
        format_str: 时间格式字符串
    
    Returns:
        格式化的时间字符串
    """
    return datetime.datetime.now().strftime(format_str)

def safe_divide(a, b):
    """
    安全除法，避免除零错误
    
    Args:
        a: 被除数
        b: 除数
    
    Returns:
        除法结果或None
    """
    try:
        return a / b if b != 0 else None
    except TypeError:
        return None

def merge_dicts(*dicts):
    """
    合并多个字典
    
    Args:
        *dicts: 要合并的字典
    
    Returns:
        合并后的字典
    """
    result = {}
    for d in dicts:
        if isinstance(d, dict):
            result.update(d)
    return result 